# -*- coding: utf-8 -*-
"""
@Time    : 2024/7/13 15:48 
@Author  : ZhangShenao 
@File    : 1.使用Chain.py 
@Desc    : 使用Chain进行调用
"""
import os

import dotenv
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import ChatOpenAI

# 加载环境变量
dotenv.load_dotenv()

# 编排Prompt
prompt = ChatPromptTemplate.from_template('{flower}的花语是？')

# 创建LLM
llm = ChatOpenAI(model_name='gpt-3.5-turbo', openai_api_base=os.getenv('OPENAI_API_BASE'))

# 创建OutputParser
parser = StrOutputParser()

# 构造Chain
chain = prompt | llm | parser

# 执行Chain,获取结果
output = chain.invoke({'flower': '满天星'})
print(output)
